Skip to content

fix(lora): restore checker-scrambled base weights once - #2520

Open
Arist12 wants to merge 2 commits into
radixark:mainfrom
Arist12:fix/lora-disaggregated-base-weight-check
Open

fix(lora): restore checker-scrambled base weights once#2520
Arist12 wants to merge 2 commits into
radixark:mainfrom
Arist12:fix/lora-disaggregated-base-weight-check

Conversation

@Arist12

@Arist12 Arist12 commented Aug 13, 2026

Copy link
Copy Markdown
Contributor

Part of #2705.

Problem

--check-weight-update-equal randomizes rollout-engine base weights at startup. LoRA normally skips frozen-base synchronization, so the first check sees a corrupted base; repeatedly sending the base is unnecessary. Multi-LoRA slot parameters must also stay out of the base converter.

Change

Restore the frozen base only on the first checked update for tensor and distributed transports, then return to adapter-only updates. Recognize both .adapter. and .adapters.<slot>. parameter names when filtering the base path.

Bridge-built architectures whose base names require Bridge conversion must include #2683 before this forced base sync is exercised; the Qwen2.5 validation below does not cover that dependency.

Validation

  • 34 focused fast tests covering two updates and single/Multi-LoRA names
  • 4x MI350X disaggregated checker E2E, including a rollout after the first optimizer update

@Arist12
Arist12 force-pushed the fix/lora-disaggregated-base-weight-check branch 5 times, most recently from 468cad9 to b2e48cb Compare August 17, 2026 04:50
@Arist12
Arist12 marked this pull request as ready for review August 17, 2026 04:54

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@Arist12
Arist12 marked this pull request as draft August 17, 2026 06:47
@Arist12
Arist12 marked this pull request as ready for review August 19, 2026 01:19

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@Arist12
Arist12 marked this pull request as draft August 19, 2026 01:44
@Arist12 Arist12 changed the title fix(lora): sync base weights on the distributed path when the weight checker is armed fix(lora): sync base weights for distributed checker Aug 19, 2026
@Arist12
Arist12 marked this pull request as ready for review August 20, 2026 15:42

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@Arist12 Arist12 changed the title fix(lora): sync base weights for distributed checker fix(lora): restore checker-scrambled base weights once Aug 20, 2026
@Arist12
Arist12 force-pushed the fix/lora-disaggregated-base-weight-check branch from bf9ee36 to e4824a3 Compare August 21, 2026 18:27
@Arist12
Arist12 force-pushed the fix/lora-disaggregated-base-weight-check branch from e4824a3 to 7ecf535 Compare August 21, 2026 18:58
Registering a new 400s 4-GPU E2E is a CI-budget decision that belongs with
the reviewer, not with a weight-sync fix. The gating logic is already covered
by the fast tests in this PR, and the disaggregated transport was validated
by hand on 8x MI350X.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant